/*******************************************************************************
 * Copyright (c) 2016 - 2022 Advanced Micro Devices, Inc. All rights reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 ******************************************************************************/
#pragma once

#include <hip/hip_runtime.h>
#include <set>
#include <string>

#define HIP_CHECK(c)                                  \
  {                                                   \
    if (c != hipSuccess) {                            \
      printf("HIP Error : %s", hipGetErrorString(c)); \
      printf(" %s %d\n", __FILE__, __LINE__);         \
      exit(c);                                        \
    }                                                 \
  }

namespace fbgemm_gpu::rocm {

[[nodiscard]] inline bool is_supported_cdna() {
  const std::set<std::string> supported_archs{"gfx942", "gfx90a"};
  int device_id = 0;
  HIP_CHECK(hipGetDevice(&device_id));
  hipDeviceProp_t dev_props;
  HIP_CHECK(hipGetDeviceProperties(&dev_props, device_id));
  std::string gcn_arch = dev_props.gcnArchName;
  gcn_arch = gcn_arch.substr(0, gcn_arch.find(":"));
  return supported_archs.contains(gcn_arch);
}

} // namespace fbgemm_gpu::rocm
